import numpy as np
import matplotlib.pyplot as plt

#阶跃函数
def step_function(x):
    y = x > 0
    return y.astype(np.int_)  #布尔值转为整数0或1

#上方也可以用：
#return np.where(x >= 0, 1, 0)
#其中np.where(条件, 条件为True时的值, 条件为False时的值)

#取点
x = np.arange(-5.0,5.0,0.1)
y = step_function(x)

#绘图
plt.plot(x, y)
plt.title("Step Function")
plt.xlabel("Input")
plt.ylabel("Output")

#设置y轴的范围为（-0.1，1.1） 
plt.ylim(-0.1, 1.1)
plt.grid()
plt.show()
